【Leetcode】【python】First Missing Positive 缺失的第一个正数

题目大意

给定一个未经排序的数组,寻找第一个缺失的正整数
你的算法应该满足O(n)时间和常数空间复杂度

解题思路

参考:http://bookshadow.com/weblog/2015/09/14/leetcode-first-missing-positive/

尽可能地把数组中不大于n(n为数组长度)的正整数放置到下标+1与其数值相同的位置上

第一个下标+1与数值不同的数字,即为所求。

例如数组nums = [3,4,-1,1],调整位置后的结果为:[1,-1,3,4]

除第二个数字外,其余数字均满足nums[i] = i + 1,因此返回2

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution(object):
def firstMissingPositive(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n = len(nums)
for i in range(n):
# print nums[i], i + 1, nums[nums[i] - 1]
while nums[i] > 0 and nums[i] <= n and nums[i] != i + 1 and nums[i] != nums[nums[i] - 1]:
nums[nums[i] - 1], nums[i] = nums[i], nums[nums[i] - 1]
# print nums
for i in range(n):
if i + 1 != nums[i]:
return i + 1
return n + 1

总结

tricky题目